package com.netty.pro.netty.protobuf2;

import com.netty.pro.netty.protobuf.PeopleModel;
import io.netty.buffer.ByteBuf;
import io.netty.channel.ChannelHandlerContext;
import io.netty.channel.ChannelInboundHandlerAdapter;
import io.netty.util.CharsetUtil;

import java.util.Random;

/**
 * @author xuzn
 * @ClassName: NettyClientHandler
 * @ProjectName netty-pro
 * @date 2020/3/10/17:36
 */
public class ProtobufNettyClientHandler extends ChannelInboundHandlerAdapter {

  @Override
  public void channelRegistered(ChannelHandlerContext ctx) throws Exception {
    System.out.println("客户端注册到服务器上了~");
  }

  // 当通道就绪时就会触发该方法
  @Override
  public void channelActive(ChannelHandlerContext ctx) throws Exception {
    // 发送一个对象到服务器
    int random = new Random().nextInt(3);
    DataInfo.AllMessage allMessage = null;
    if (random == 0) {
      allMessage = DataInfo.AllMessage.newBuilder().setDataType(DataInfo.AllMessage.DataType.studentType).
              setStudent(DataInfo.Student.newBuilder().setId(2).setName("李四").build()).build();
    } else {
      allMessage = DataInfo.AllMessage.newBuilder().setDataType(DataInfo.AllMessage.DataType.workerType).
              setWoker(DataInfo.Worker.newBuilder().setAge(19).setName("大刀王五").build()).build();
    }
    ctx.writeAndFlush(allMessage);
  }

  // 当通道有读取事件时会触发
  @Override
  public void channelRead(ChannelHandlerContext ctx, Object msg) throws Exception {
    ByteBuf buf = (ByteBuf) msg;
    System.out.println("服务器回复的消息:" + buf.toString(CharsetUtil.UTF_8));
    System.out.println("服务器地址:" + ctx.channel().remoteAddress());
  }

  @Override
  public void exceptionCaught(ChannelHandlerContext ctx, Throwable cause) throws Exception {
    cause.printStackTrace();
    ctx.close();
  }
}
